# 函数练习  开发ATM小程序

# 要求：实现查询，存储，取款，退出

# 我的总金额
count = 5000


# 菜单函数
def menu():
    print("***********主菜单************")
    print("查询余额\t【1】")
    print("存款\t\t【2】")
    print("取款\t\t【3】")
    print("退出\t\t【4】")
    return input("请输入：")


# 查询余额
def query():
    print(f"您的余额为：{count}")


# 存储函数
def save(num):
    global count
    count += int(num)
    print("存储成功")


# 存储函数
def get(num):
    global count
    count -= int(num)
    print("取款成功")


while True:
    name = input("请输入账号")
    password = input("请输入密码")
    # 验证账号密码
    if name == "azemat" and password == "123456":
        while True:
            user_menu = menu()
            if user_menu == "1":
                query()
            elif user_menu == "2":
                save(input("请输入您要存款的金额"))
            elif user_menu == "3":
                get(input("请输入您要取款的金额"))
            else:
                print("输入有误，系统已退出")
                break
    else:
        print("您输入的账号密码错误")
        break
